package org.basis.netty.chat.rpc.client;

import io.netty.bootstrap.Bootstrap;
import io.netty.channel.Channel;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;
import io.netty.util.concurrent.DefaultPromise;
import lombok.extern.slf4j.Slf4j;
import org.basis.netty.chat.message.RpcRequestMessage;
import org.basis.netty.chat.protocol.MessageCodecSharable;
import org.basis.netty.chat.protocol.ProcotolFrameDecoder;
import org.basis.netty.chat.protocol.RpcResponseMessageHandler;
import org.basis.netty.chat.rpc.server.service.HelloService;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;

/**
 * rpc客户端管理器
 *
 * @author Mr_wenpan@163.com 2021/09/28 17:23
 */
@Slf4j
public class RpcClientManager {

    public static void main(String[] args) {

        HelloService proxyService = getProxyService(HelloService.class);
        log.info("返回结果：{}", proxyService.sayHello("文攀"));

    }

    /**
     * 通过接口的class创建该接口的代理对象
     * 这里直接基于JDK提供的创建动态代理的工具来创建代理对象
     *
     * @param serviceClass 接口的class
     * @return T 代理对象
     * @author Mr_wenpan@163.com 2021/9/28 5:25 下午
     */
    public static <T> T getProxyService(Class<T> serviceClass) {

        // 该接口的Class对象是被那个类加载器加载的
        ClassLoader classLoader = serviceClass.getClassLoader();
        // 获取到该接口所有的interface
        Class<?>[] interfaces = {serviceClass};

        // jdk代理必须的handler，代理对象的方法执行就会调用这里的invoke方法。自动传入调用的方法 + 方法参数
        InvocationHandler invocationHandler = new InvocationHandler() {
            @Override
            public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

                // 1、将方法调用转换为rpc请求消息
                // 客户端调用接口的方法的时候
                int sequenceId = SequenceIdGenerator.nextId();
                // 封装RPC请求消息
                RpcRequestMessage rpcRequestMessage = new RpcRequestMessage(sequenceId,
                        serviceClass.getName(),
                        method.getName(),
                        method.getReturnType(),
                        method.getParameterTypes(),
                        args);

                // 2、将消息对象发送出去
                getChannel().writeAndFlush(rpcRequestMessage);

                // 3、准备一个空的promise对象来接收server返回的结果
                // 指定promise对象异步接收结果的线程，这里使用发送消息的channel的线程来接收消息（getChannel().eventLoop()）
                DefaultPromise<Object> promise = new DefaultPromise<>(getChannel().eventLoop());

                // 将这个promise缓存起来，以便于server响应结果回来的时候能够通过消息的sequenceId正确的找到这个promise
                // 那么这个promise在哪里接收的结果呢？当然是在响应handler里，因为只有当server端响应了promise才应该被设置值
                RpcResponseMessageHandler.PROMISES.put(sequenceId, promise);


                // 异步等待的方式
//                promise.addListener(future -> {
//                    // 线程
//                });

                // 同步等待的方式(按道理这里应该加超时等待，不能永远无休止的等待)
                // 调用接口的线程等待，直到promise有结果（正常或异常）
                promise.await();

                // server返回结果后结束上面的阻塞，执行这里
                if (promise.isSuccess()) {
                    // 正常调用
                    return promise.getNow();
                } else {
                    // 异常调用
                    throw new RuntimeException(promise.cause());
                }

            }
        };

        Object proxy = Proxy.newProxyInstance(classLoader, interfaces, invocationHandler);

        // 返回代理对象
        return (T) proxy;
    }

    /**
     * 发送数据的channel
     */
    private static Channel channel = null;
    private static final Object LOCK = new Object();

    /**
     * 获取唯一的 channel 对象(需要通过这个channel将数据发送给server端)
     */
    public static Channel getChannel() {

        if (channel != null) {
            return channel;
        }
        // 双从检查
        synchronized (LOCK) {
            if (channel != null) {
                return channel;
            }
            initChannel();
            return channel;
        }
    }

    /**
     * 初始化 channel 方法
     */
    private static void initChannel() {
        NioEventLoopGroup group = new NioEventLoopGroup();
        LoggingHandler LOGGING_HANDLER = new LoggingHandler(LogLevel.DEBUG);
        MessageCodecSharable MESSAGE_CODEC = new MessageCodecSharable();
        RpcResponseMessageHandler RPC_HANDLER = new RpcResponseMessageHandler();
        Bootstrap bootstrap = new Bootstrap();
        bootstrap.channel(NioSocketChannel.class);
        bootstrap.group(group);
        bootstrap.handler(new ChannelInitializer<SocketChannel>() {
            @Override
            protected void initChannel(SocketChannel ch) throws Exception {
                ch.pipeline().addLast(new ProcotolFrameDecoder());
                ch.pipeline().addLast(LOGGING_HANDLER);
                ch.pipeline().addLast(MESSAGE_CODEC);
                ch.pipeline().addLast(RPC_HANDLER);
            }
        });
        try {
            channel = bootstrap.connect("localhost", 8080).sync().channel();
            channel.closeFuture().addListener(future -> {
                group.shutdownGracefully();
            });
        } catch (Exception e) {
            log.error("client error", e);
        }
    }

}
